# -*- coding: utf-8 -*-
"""
Created on Wed Oct 31 20:08:11 2018

@author: 陈忠涛
"""
# 作死小体验：求平安保险在2017年每个月股票收盘价的平均值。
# 使用了Anaconda解释库（里面集成了大量科学包，numpy、pandas， SciPy library...）

import tushare as ts
import time
from dateutil.parser import parse
import matplotlib.pyplot as plt

print('2018年平安保险股票日线行情')
pro = ts.pro_api('709417260d6eb8e46e32df28c6bee0709b4d4eec157d13ff42d7faed') # 初始化pro接口
data = pro.daily(ts_code='000001.SZ', start_date='20180101', end_date='20181231')
data.index = range(1,len(data) + 1)
new_data = data.drop(['open', 'high', 'low', 'pre_close', 'change', 'pct_change', 'vol', 'amount'], axis = 1)
print(new_data)
month_list = new_data['trade_date']
new_month = []
for m in month_list:
    tra_mon = time.strptime(str(parse(m)), '%Y-%m-%d %H:%M:%S').tm_mon
    new_month.append(tra_mon)
new_data['月份'] = new_month
# m = new_data.groupby('month').count() # 求出每个月的交易次数
m = new_data.groupby('月份').mean()
m.columns = ['月股票收盘价平均值']
print('2018年平安保险每个月股票平均收盘价')
print(m)

# 数据绘图

x = m.index
y = m.values


# 折线图
print(plt.plot(x, y, 'o--'))

# 散点图
# print(plt.plot(x,y,'o'))

# 柱状图
# print(plt.bar(x,y))

